<?php
/**
 * User: 周星
 * Date: 2019/6/26
 * Time: 15:54
 * Email: zhouxing@benbang.com
 */

namespace util;


/**
 * * Zip 文件包工具
 * *
 * * @author wengxianhu
 * * @date 2013-08-05
 * */
class Zip {

    protected $zip;
    protected $root;
    protected $ignored_names;

    public function __construct () {
        $this->zip = new \ZipArchive();
    }

    /**
     * * 解压zip文件到指定文件夹
     * *
     * * @access public
     * * @param string $zipfile 压缩文件路径
     * * @param string $path 压缩包解压到的目标路径
     * * @return boolean 解压成功返回 true 否则返回 false
     * */
    public function unzip ($zipfile, $path) {
        if ($this->zip->open($zipfile) === true) {
            $file_tmp = @fopen($zipfile, "rb");
            $bin = fread($file_tmp, 15); //只读15字节 各个不同文件类型，头信息不一样。
            fclose($file_tmp);
            /* 只针对zip的压缩包进行处理 */
            if (true === $this->getTypeList($bin)) {
                $result = $this->zip->extractTo($path);
                $this->zip->close();

                return $result;
            } else {
                return false;
            }
        }

        return false;
    }

    /**
     * * 创建压缩文件
     * *
     * * @access public
     * * @param string $zipfile 将要生成的压缩文件路径
     * * @param string|array $folder 将要被压缩的文件夹路径
     * * @param array $ignored 要忽略的文件列表
     * * @return boolean 压缩包生成成功返回true 否则返回 false
     * */
    public function zip ($zipfile, $folder, $ignored = null) {
        $this->ignored_names = is_array($ignored) ? $ignored : $ignored ? [$ignored] : [];
        if ($this->zip->open($zipfile, \ZIPARCHIVE::CREATE) !== true) {
            return false;
        }
        if (is_dir($folder)) {
            $folder = substr($folder, - 1) == '/' ? substr($folder, 0, strlen($folder) - 1) : $folder;
            if (strstr($folder, '/')) {
                $this->root = substr($folder, 0, strrpos($folder, '/') + 1);
                $folder = substr($folder, strrpos($folder, '/') + 1);
            }
            $this->createZip($folder);
        } elseif (is_file($folder)) {
            $this->addFile($folder);
        } elseif (is_array($folder)) {
            foreach ($folder as $v) {
                $this->addFile($folder);
            }
        } else {
            return false;
        }

        return $this->zip->close();
    }

    /**
     * * 递归添加文件到压缩包
     * *
     * * @access private
     * * @param string $folder 添加到压缩包的文件夹路径
     * * @param string $parent 添加到压缩包的文件夹上级路径
     * * @return void
     * */
    private function createZip ($folder, $parent=null) {
        $full_path = $this->root . $parent . $folder;
        $zip_path = $parent . $folder;
        $this->zip->addEmptyDir($zip_path);
        $dir = new \DirectoryIterator($full_path);
        foreach($dir as $file) {
            if(!$file->isDot()) {
                $filename = $file->getFilename();
                if(!in_array($filename, $this->ignored_names)) {
                    if($file->isDir()) {
                        $this->createZip($filename, $zip_path.'/');
                    }else {
                        $this->zip->addFile($full_path.'/'.$filename, $zip_path.'/'.$filename);
                    }
                }
            }
        }
    }

    /**
     * * 添加压缩文件
     * *
     * * @access private
     * * @param string $filePath 添加到压缩包的文件夹路径
     * * @return void
     * */
    private function addFile ($filePath) {
        if (is_file($filePath)) {
            $fileName =  substr($filePath, strrpos($filePath, '/') + 1);
            $this->zip->addFile($filePath, $fileName);
        }
    }

    /**
     * * 读取压缩包文件与目录列表
     * *
     * * @access public
     * * @param string $zipfile 压缩包文件
     * * @return array 文件与目录列表
     * */
    public function fileList ($zipfile) {
        $file_dir_list = [];
        $file_list = [];
        if ($this->zip->open($zipfile) == true) {
            for ($i = 0; $i < $this->zip->numFiles; $i ++) {
                $numfiles = $this->zip->getNameIndex($i);
                if (preg_match('/\/$/i', $numfiles)) {
                    $file_dir_list[] = $numfiles;
                } else {
                    $file_list[] = $numfiles;
                }
            }
        }

        return ['files' => $file_list, 'dirs' => $file_dir_list];
    }

    /**
     * * 得到文件头与文件类型映射表
     * *
     * * @param $bin string 文件的二进制前一段字符
     * * @return boolean
     * *@author wengxianhu
     * * @date 2013-08-10
     * */
    private function getTypeList ($bin) {
        $array = [
            ["504B0304", "zip"],
        ];
        foreach ($array as $v) {
            $blen = strlen(pack("H*", $v[0])); //得到文件头标记字节数
            $tbin = substr($bin, 0, intval($blen)); ///需要比较文件头长度
            if (strtolower($v[0]) == strtolower(array_shift(unpack("H*", $tbin)))) {
                return true;
            }

        }

        return false;
    }
}